package tips.p_others;

import java.util.Arrays;

/**
 * 元素的 频数 是该元素在一个数组中出现的次数。
 * 给你一个整数数组 nums 和一个整数 k 。在一步操作中，你可以选择 nums 的一个下标，并将该下标对应元素的值增加 1 。
 * 执行最多 k 次操作后，返回数组中最高频元素的 最大可能频数 。
 * <p>示例 1：
 * 输入：nums = [1,2,4], k = 5
 * 输出：3
 * 解释：对第一个元素执行 3 次递增操作，对第二个元素执 2 次递增操作，此时 nums = [4,4,4] 。
 * 4 是数组中最高频元素，频数是 3 。
 * <p>示例 2：
 * 输入：nums = [1,4,8,13], k = 5
 * 输出：2
 * 解释：存在多种最优解决方案：
 * - 对第一个元素执行 3 次递增操作，此时 nums = [4,4,8,13] 。4 是数组中最高频元素，频数是 2 。
 * - 对第二个元素执行 4 次递增操作，此时 nums = [1,8,8,13] 。8 是数组中最高频元素，频数是 2 。
 * - 对第三个元素执行 5 次递增操作，此时 nums = [1,4,13,13] 。13 是数组中最高频元素，频数是 2 。
 * <p>示例 3：
 * 输入：nums = [3,9,6], k = 2
 * 输出：1
 * <p>提示：
 * 1 <= nums.length <= 10^5
 * 1 <= nums[i] <= 10^5
 * 1 <= k <= 10^5
 *
 * @author hc
 */
public class Demo1838 {

    public int maxFrequency(int[] nums, int k) {
        // 滑动窗口 + 前缀和
        if (nums.length == 1) {
            return 1;
        }
        Arrays.sort(nums);
        int l = 0, r = 0, maxFreq = 1;
        int sum = 0, len = nums.length;
        while (r < len) {
            sum += nums[r];
            if (k >= nums[r] * (r - l + 1) - sum) {
                maxFreq = Math.max(maxFreq, r - l + 1);
            } else {
                sum -= nums[l];
                ++l;
            }
            ++r;
        }
        return maxFreq;
    }

    public static void main(String[] args) {
        System.out.println(new Demo1838().maxFrequency(new int[]{1, 2, 4}, 5));
        System.out.println(new Demo1838().maxFrequency(new int[]{1,4,8,13}, 5));
    }
}
